#include "oled.h"
#include "i2c.h"
#include "OLED_Font.h"


#define OLED_SET_MODE(mode) ((mode) ? 0x40 : 0x00)
#define OLED_ADDRESS 0x78

//发送一个字节
//mode:数据/命令标志 0,表示命令;1,表示数据;
void OLED_WR_Byte(uint8_t dat,uint8_t mode)
{
	uint8_t sendBuff[2];
	sendBuff[0] = OLED_SET_MODE(mode);
	sendBuff[1] = dat;
	HAL_I2C_Master_Transmit(&hi2c1, OLED_ADDRESS, (uint8_t *)sendBuff, 2, HAL_MAX_DELAY);
}
//反显函数
void OLED_ColorTurn(uint8_t i)
{
	if(i==0)
		{
			OLED_WR_Byte(0xA6,OLED_CMD);//正常显示
		}
	if(i==1)
		{
			OLED_WR_Byte(0xA7,OLED_CMD);//反色显示
		}
}

//屏幕旋转180度
void OLED_DisplayTurn(uint8_t i)
{
	if(i==0)
		{
			OLED_WR_Byte(0xC8,OLED_CMD);//正常显示
			OLED_WR_Byte(0xA1,OLED_CMD);
		}
	if(i==1)
		{
			OLED_WR_Byte(0xC0,OLED_CMD);//反转显示
			OLED_WR_Byte(0xA0,OLED_CMD);
		}
}

//开启OLED显示 
void OLED_DisPlay_On(void)
{
	OLED_WR_Byte(0x8D,OLED_CMD);//电荷泵使能
	OLED_WR_Byte(0x14,OLED_CMD);//开启电荷泵
	OLED_WR_Byte(0xAF,OLED_CMD);//点亮屏幕
}
//关闭OLED显示 
void OLED_DisPlay_Off(void)
{
	OLED_WR_Byte(0x8D,OLED_CMD);//电荷泵使能
	OLED_WR_Byte(0x10,OLED_CMD);//关闭电荷泵
	OLED_WR_Byte(0xAE,OLED_CMD);//关闭屏幕
}
/**
  * @brief  OLED写命令
  * @param  Command 要写入的命令
  * @retval 无
  */
void OLED_WriteCommand(uint8_t Command)
{
    uint8_t sendBuff[2];
    sendBuff[0] = 0x00;
    sendBuff[1] = Command;
    HAL_I2C_Master_Transmit(&hi2c1, OLED_ADDRESS, (uint8_t *)sendBuff, 2, HAL_MAX_DELAY);
}

/**
  * @brief  OLED写数据
  * @param  Data 要写入的数据
  * @retval 无
  */
void OLED_WriteData(uint8_t Data)
{
    uint8_t sendBuff[2];
    sendBuff[0] = 0x40;
    sendBuff[1] = Data;
    HAL_I2C_Master_Transmit(&hi2c1, OLED_ADDRESS, (uint8_t *)sendBuff, 2, HAL_MAX_DELAY);   
}
/**
  * @brief  OLED设置光标位置
  * @param  Y 以左上角为原点，向下方向的坐标，范围：0~7
  * @param  X 以左上角为原点，向右方向的坐标，范围：0~127
  * @retval 无
  */
void OLED_SetCursor(uint8_t Y, uint8_t X)
{
	OLED_WriteCommand(0xB0 | Y);					//设置Y位置
	OLED_WriteCommand(0x10 | ((X & 0xF0) >> 4));	//设置X位置高4位
	OLED_WriteCommand(0x00 | (X & 0x0F));			//设置X位置低4位
}
/**
  * @brief  OLED清屏
  * @param  无
  * @retval 无
  */
void OLED_Clear(void)
{  
	uint8_t i, j;
	for (j = 0; j < 8; j++)
	{
		OLED_SetCursor(j, 0);
		for(i = 0; i < 128; i++)
		{
			OLED_WriteData(0x00);
		}
	}
}
/**
  * @brief  OLED显示一个字符
  * @param  Line 行位置，范围：1~4
  * @param  Column 列位置，范围：1~16
  * @param  Char 要显示的一个字符，范围：ASCII可见字符
  * @retval 无
  */
void OLED_ShowChar(uint8_t Line, uint8_t Column, char Char)
{      	
	uint8_t i;
	OLED_SetCursor((Line - 1) * 2, (Column - 1) * 8);		//设置光标位置在上半部分
	for (i = 0; i < 8; i++)
	{
		OLED_WriteData(OLED_F8x16[Char - ' '][i]);			//显示上半部分内容
	}
	OLED_SetCursor((Line - 1) * 2 + 1, (Column - 1) * 8);	//设置光标位置在下半部分
	for (i = 0; i < 8; i++)
	{
		OLED_WriteData(OLED_F8x16[Char - ' '][i + 8]);		//显示下半部分内容
	}
}

/**
  * @brief  OLED显示字符串
  * @param  Line 起始行位置，范围：1~4
  * @param  Column 起始列位置，范围：1~16
  * @param  String 要显示的字符串，范围：ASCII可见字符
  * @retval 无
  */
void OLED_ShowString(uint8_t Line, uint8_t Column, char *String)
{
	uint8_t i;
	for (i = 0; String[i] != '\0'; i++)
	{
		OLED_ShowChar(Line, Column + i, String[i]);
	}
}
/**
  * @brief  OLED次方函数
  * @retval 返回值等于X的Y次方
  */
uint32_t OLED_Pow(uint32_t X, uint32_t Y)
{
	uint32_t Result = 1;
	while (Y--)
	{
		Result *= X;
	}
	return Result;
}
/**
  * @brief  OLED显示数字（十进制，正数）
  * @param  Line 起始行位置，范围：1~4
  * @param  Column 起始列位置，范围：1~16
  * @param  Number 要显示的数字，范围：0~4294967295
  * @param  Length 要显示数字的长度，范围：1~10
  * @retval 无
  */
void OLED_ShowNum(uint8_t Line, uint8_t Column, uint32_t Number, uint8_t Length)
{
	uint8_t i;
	for (i = 0; i < Length; i++)							
	{
		OLED_ShowChar(Line, Column + i, Number / OLED_Pow(10, Length - i - 1) % 10 + '0');
	}
}
/**
  * @brief  OLED显示数字（十进制，带符号数）
  * @param  Line 起始行位置，范围：1~4
  * @param  Column 起始列位置，范围：1~16
  * @param  Number 要显示的数字，范围：-2147483648~2147483647
  * @param  Length 要显示数字的长度，范围：1~10
  * @retval 无
  */
void OLED_ShowSignedNum(uint8_t Line, uint8_t Column, int32_t Number, uint8_t Length)
{
	uint8_t i;
	uint32_t Number1;
	if (Number >= 0)
	{
		OLED_ShowChar(Line, Column, '+');
		Number1 = Number;
	}
	else
	{
		OLED_ShowChar(Line, Column, '-');
		Number1 = -Number;
	}
	for (i = 0; i < Length; i++)							
	{
		OLED_ShowChar(Line, Column + i + 1, Number1 / OLED_Pow(10, Length - i - 1) % 10 + '0');
	}
}
/**
  * @brief  OLED显示数字（十六进制，正数）
  * @param  Line 起始行位置，范围：1~4
  * @param  Column 起始列位置，范围：1~16
  * @param  Number 要显示的数字，范围：0~0xFFFFFFFF
  * @param  Length 要显示数字的长度，范围：1~8
  * @retval 无
  */
void OLED_ShowHexNum(uint8_t Line, uint8_t Column, uint32_t Number, uint8_t Length)
{
	uint8_t i, SingleNumber;
	for (i = 0; i < Length; i++)							
	{
		SingleNumber = Number / OLED_Pow(16, Length - i - 1) % 16;
		if (SingleNumber < 10)
		{
			OLED_ShowChar(Line, Column + i, SingleNumber + '0');
		}
		else
		{
			OLED_ShowChar(Line, Column + i, SingleNumber - 10 + 'A');
		}
	}
}
/**
  * @brief  OLED显示数字（二进制，正数）
  * @param  Line 起始行位置，范围：1~4
  * @param  Column 起始列位置，范围：1~16
  * @param  Number 要显示的数字，范围：0~1111 1111 1111 1111
  * @param  Length 要显示数字的长度，范围：1~16
  * @retval 无
  */
void OLED_ShowBinNum(uint8_t Line, uint8_t Column, uint32_t Number, uint8_t Length)
{
	uint8_t i;
	for (i = 0; i < Length; i++)							
	{
		OLED_ShowChar(Line, Column + i, Number / OLED_Pow(2, Length - i - 1) % 2 + '0');
	}
}
/**
  * @brief  OLED初始化
  * @param  无
  * @retval 无
  */
void OLED_Init(void)
{
	uint32_t i, j;
	
	for (i = 0; i < 1000; i++)			//上电延时
	{
		for (j = 0; j < 1000; j++);
	}
	
	// OLED_I2C_Init();			//端口初始化
	
	OLED_WriteCommand(0xAE);	//关闭显示
	
	OLED_WriteCommand(0xD5);	//设置显示时钟分频比/振荡器频率
	OLED_WriteCommand(0x80);
	
	OLED_WriteCommand(0xA8);	//设置多路复用率
	OLED_WriteCommand(0x3F);
	
	OLED_WriteCommand(0xD3);	//设置显示偏移
	OLED_WriteCommand(0x00);
	
	OLED_WriteCommand(0x40);	//设置显示开始行
	
	OLED_WriteCommand(0xA1);	//设置左右方向，0xA1正常 0xA0左右反置
	
	OLED_WriteCommand(0xC8);	//设置上下方向，0xC8正常 0xC0上下反置

	OLED_WriteCommand(0xDA);	//设置COM引脚硬件配置
	OLED_WriteCommand(0x12);
	
	OLED_WriteCommand(0x81);	//设置对比度控制
	OLED_WriteCommand(0xCF);

	OLED_WriteCommand(0xD9);	//设置预充电周期
	OLED_WriteCommand(0xF1);

	OLED_WriteCommand(0xDB);	//设置VCOMH取消选择级别
	OLED_WriteCommand(0x30);

	OLED_WriteCommand(0xA4);	//设置整个显示打开/关闭

	OLED_WriteCommand(0xA6);	//设置正常/倒转显示

	OLED_WriteCommand(0x8D);	//设置充电泵
	OLED_WriteCommand(0x14);

	OLED_WriteCommand(0xAF);	//开启显示
		
	OLED_Clear();				//OLED清屏
}


//uint8_t u8g2_gpio_and_delay_stm32(U8X8_UNUSED u8x8_t *u8x8, U8X8_UNUSED uint8_t msg, U8X8_UNUSED uint8_t arg_int, U8X8_UNUSED void *arg_ptr)
//{
//  switch (msg)
//  {
//  // Initialize SPI peripheral
//  case U8X8_MSG_GPIO_AND_DELAY_INIT:
//    /* HAL initialization contains all what we need so we can skip this part. */
//    // GPIO_InitTypeDef GPIO_InitStruct = {0};
//    // GPIO_InitStruct.Pin = GPIO_PIN_6|GPIO_PIN_7;
//    // GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_PP;
//    // GPIO_InitStruct.Pull = GPIO_PULLUP;
//    // GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_VERY_HIGH;
//    // HAL_GPIO_Init(GPIOB, &GPIO_InitStruct);
//    break;
//  // Function which implements a delay, arg_int contains the amount of ms
//  case U8X8_MSG_DELAY_MILLI:
//    HAL_Delay(arg_int);

//    break;
//  // Function which delays 10us
//  case U8X8_MSG_DELAY_10MICRO:
//    for (uint16_t n = 0; n < 320; n++)
//    {
//      __NOP();
//    }

//    break;
//  // Function which delays 100ns
//  case U8X8_MSG_DELAY_100NANO:
//    __NOP();
//    break;

//  case U8X8_MSG_GPIO_I2C_CLOCK:
//  //   if (arg_int)
//  //     HAL_GPIO_WritePin(GPIOB, GPIO_PIN_6, GPIO_PIN_SET);
//  //   else
//  //     HAL_GPIO_WritePin(GPIOB, GPIO_PIN_6, GPIO_PIN_RESET);
//  //   break;
//  // case U8X8_MSG_GPIO_I2C_DATA:
//  //   if (arg_int)
//  //     HAL_GPIO_WritePin(GPIOB, GPIO_PIN_7, GPIO_PIN_SET);
//  //   else
//  //     HAL_GPIO_WritePin(GPIOB, GPIO_PIN_7, GPIO_PIN_RESET);
//  //   break;
//  default:
//    return 0; // A message was received which is not implemented, return 0 to indicate an error
//  }

//  return 1; // command processed successfully.
//}
//uint8_t u8x8_byte_hw_i2c(u8x8_t *u8x8, uint8_t msg, uint8_t arg_int, void *arg_ptr)
//{
//  static uint8_t buffer[32]; /* u8g2/u8x8 will never send more than 32 bytes between START_TRANSFER and END_TRANSFER */
//  static uint8_t buf_idx;
//  uint8_t *data;

//  switch (msg)
//  {
//  case U8X8_MSG_BYTE_INIT:
//    /* add your custom code to init i2c subsystem */
//		MX_I2C1_Init();
//    break;
//  case U8X8_MSG_BYTE_SET_DC:
//	break;
//  case U8X8_MSG_BYTE_SEND:
//    data = (uint8_t *)arg_ptr;
//    while (arg_int > 0)
//    {
//      buffer[buf_idx++] = *data;
//      data++;
//      arg_int--;
//    }
//    break;
//  case U8X8_MSG_BYTE_START_TRANSFER:
//    buf_idx = 0;
//    break;
//  case U8X8_MSG_BYTE_END_TRANSFER:
//      HAL_I2C_Master_Transmit(&hi2c1, u8x8_GetI2CAddress(u8x8), buffer, buf_idx, 1000);

//    // HAL_I2C_Mem_Write(&hi2c1, u8x8_GetI2CAddress(u8x8), u8x8_GetI2CAddress(u8x8), buf_idx, (uint8_t *)buffer, buf_idx, 1000);
//    break;
//  default:
//    return 0;
//  }
//  return 1;
//}

//void u8g2_Init(u8g2_t *u8g2) {
//    u8g2_Setup_ssd1306_i2c_128x64_noname_f(u8g2, U8G2_R0, u8x8_byte_hw_i2c, u8g2_gpio_and_delay_stm32);  // 初始化 u8g2 结构体
//    u8g2_InitDisplay(u8g2);      // 根据所选的芯片进行初始化工作，初始化完成后，显示器处于关闭状态
//    u8g2_SetPowerSave(u8g2, 0);  // 打开显示器
//    u8g2_ClearBuffer(u8g2);
//}


//void draw(u8g2_t *u8g2)
//{
//	u8g2_ClearBuffer(u8g2); 
//   u8g2_SetFontMode(u8g2, 1);
//   u8g2_SetFontDirection(u8g2, 0); 
//   u8g2_SetFont(u8g2, u8g2_font_inb24_mf); 
//   u8g2_DrawStr(u8g2, 0, 20, "U");

//   u8g2_SetFontDirection(u8g2, 1);
//   u8g2_SetFont(u8g2, u8g2_font_inb30_mn);
//   u8g2_DrawStr(u8g2, 21,8,"8");

//   u8g2_SetFontDirection(u8g2, 0);
//   u8g2_SetFont(u8g2, u8g2_font_inb24_mf);
//   u8g2_DrawStr(u8g2, 51,30,"g");
//   u8g2_DrawStr(u8g2, 67,30,"\xb2");

//   u8g2_DrawHLine(u8g2, 2, 35, 47);
//   u8g2_DrawHLine(u8g2, 3, 36, 47);
//   u8g2_DrawVLine(u8g2, 45, 32, 12);
//   u8g2_DrawVLine(u8g2, 46, 33, 12);

//   u8g2_SetFont(u8g2, u8g2_font_4x6_tr);
//   u8g2_DrawStr(u8g2, 1,54,"github.com/olikraus/u8g2");
//	 u8g2_SendBuffer(u8g2);
//}

